#!/usr/bin/python
#-*- coding: utf-8 -*-



import sys
reload(sys)

import urllib2
import urllib
import re

import ssl
ssl._create_default_https_context = ssl._create_unverified_context

try:
    import json
except ImportError:
    import simplejson as json



class SaltAPI(object):
    __tocken_id = ''
    def __init__(self, url, username, password):
        self.__url = url.rstrip('/')  #salt-api监控的地址和端口
        self.__user = username
        self.__password = password
        #self.__token_id = self.tocken_id


    def tocken_id(self):
        ''' user login and get token id '''
        params = {'eauth': 'pam', 'username': self.__user, 'password': self.__password}
        encode = urllib.urlencode(params)
        obj = urllib.unquote(encode)
        content = self.postRequest(obj, prefix='/login')
        try:
            self.__tocken_id = content['return'][0]['token']
        except KeyError:
            raise KeyError

    def postRequest(self, obj, prefix='/'):
        url = self.__url + prefix
        headers = {'X-Auth-Token': self.__tocken_id}
        print headers  ## 获取salt-api头部tocken的信息
        req = urllib2.Request(url, obj, headers)
        print req
        opener = urllib2.urlopen(req)
        content = json.loads(opener.read())
        return content

    def list_all_key(self):
        params = {'client':'wheel', 'fun': 'key.list_all'}
        obj = urllib.urlencode(params)
        self.tocken_id()
        content = self.postRequest(obj)
        minions = content['return'][0]['data']['return']['minions']
        minions_pre = content['return'][0]['data']['return']['minions_pre']
        return minions, minions_pre

    def delete_key(self, node_name):
        params = {'client': 'wheel', 'fun': 'key.delete', 'match': node_name}
        obj = urllib.urlencode(params)
        self.tocken_id()
        content = self.postRequest(obj)
        ret = content['return'][0]['data']['success']
        return ret

    def accept_key(self, node_name):
        params = {'client': 'wheel', 'fun': 'key.accept', 'match': node_name}
        obj = urllib.urlencode(params)
        self.tocken_id()
        content = self.postRequest(obj)
        ret = content['return'][0]['data']['success']
        return ret

    def remote_noarg_execution(self, tgt, fun):
        ''' Execute commands without parameters '''
        params = {'client': 'local', 'tgt': tgt, 'fun': fun}
        obj = urllib.urlencode(params)
        self.tocken_id()
        content = self.postRequest(obj)
        ret = content['return'][0][tgt]
        return ret

    def remote_execution(self, tgt, fun, arg):
        ''' Command execution with parameters '''
        params = {'client': 'local', 'tgt': tgt, 'fun': fun, 'arg': arg}
        obj = urllib.urlencode(params)
        self.tocken_id()
        content = self.postRequest(obj)
        ret = content['return']
        return ret


    def target_remote_execution(self, tgt, fun, arg):
        ''' Use targeting for remote execution '''
        params = {'client': 'local', 'tgt': tgt, 'fun': fun, 'arg': arg, 'expr_form': 'nodegroup'}
        obj = urllib.urlencode(params)
        self.tocken_id()
        content = self.postRequest(obj)
        jid = content['return'][0]['jid']
        return jid

    def deploy(self, tgt, arg):
        ''' Module deployment '''
        params = {'client': 'local', 'tgt': tgt, 'fun': 'discovery.Discovery', 'arg': arg}
        obj = urllib.urlencode(params)
        self.tocken_id()
        content = self.postRequest(obj)
        return content

    def async_deploy(self, tgt, arg):
        ''' Asynchronously send a command to connected minions '''
        params = {'client': 'local_async', 'tgt': tgt, 'fun': 'discovery.Discovery', 'arg': arg}
        obj = urllib.urlencode(params)
        self.tocken_id()
        content = self.postRequest(obj)
        jid = content['return'][0]['jid']
        return jid

    def target_deploy(self, tgt, arg):
        ''' Based on the node group forms deployment '''
        params = {'client': 'local_async', 'tgt': tgt, 'fun': 'discovery.Discovery', 'arg': arg, 'expr_form': 'nodegroup'}
        obj = urllib.urlencode(params)
        self.tocken_id()
        content = self.postRequest(obj)
        jid = content['return'][0]['jid']
        return jid



def main():
    sapi = SaltAPI(url='https://192.168.0.12:8000', username='saltapi', password='saltapi')
    ##print sapi.remote_noarg_execution('client', 'discovery.Discovery')
    #print sapi.remote_noarg_execution('Home-Lnmp', 'grains.items')
    #print sapi.deploy('Home-Lnmp', 'nginx')
    #params = {'client':'local', 'fun':'discovery.Discovery', 'tgt':'*'}
    #test = spi.saltCmd(params)
    #resp = json.dumps(test, indent=1)
    #print "resp=", resp

if __name__ == '__main__':
    main()


